/**
 * Copyright [2019] [LiBo/Alex of copyright liboware@gmail.com ]
 * <p>
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 * <p>
 * http://www.apache.org/licenses/LICENSE-2.0
 * <p>
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 */
package com.hyts.generate.mybatisplus.handler;

import com.baomidou.mybatisplus.generator.config.DataSourceConfig;
import com.google.common.base.Preconditions;
import com.hyts.generate.mybatisplus.bean.AttributeDataSourceBean;
import com.hyts.generate.mybatisplus.bean.GenerateBean;
import com.hyts.generate.mybatisplus.handler.base.AbstractAttributeValueHandler;

import java.util.Objects;

/**
 * @project-name:standard-boot
 * @package-name:com.hyts.standard.boot.generator.handler
 * @author:LiBo/Alex
 * @create-date:2019-07-26 15:13
 * @copyright:libo-alex4java
 * @email:liboware@gmail.com,
 * @description:
 */
public class AttributeDataSourceHandler extends AbstractAttributeValueHandler<AttributeDataSourceBean> {

    /**
     * 构造器服务
     * @param generateBean
     */
    public AttributeDataSourceHandler(GenerateBean generateBean) {
        super(generateBean);
        Objects.requireNonNull(generateBean,"不可传递为空的数据参数对象！generateBean对象");
    }

    /**
     * 构建参数对象构建服务-数据源
     * @param param
     * @return
     */
    @Override
    public GenerateBean build(AttributeDataSourceBean param) {
        //判断条件开始
        //自动检查机制-lombok服务
        Preconditions.checkArgument(Objects.isNull(param),"传入的参数AttributeDataSourceBean不可以为空!");
        DataSourceConfig dataSourceConfig = new DataSourceConfig();
        dataSourceConfig.setDbType(param.getDbType());
        dataSourceConfig.setDriverName(param.getDriverClass());
        dataSourceConfig.setPassword(param.getPassword());
        dataSourceConfig.setUrl(param.getUrl());
        dataSourceConfig.setUsername(param.getUserName());
        generateBean.setDataSourceConfig(dataSourceConfig);
        return generateBean;
    }
}
